Replace Words with Stems — Interview Query python problem