Remove itertools import imap