PEP8, Removed some unused import

Change-Id: I1a73f84762dbb786bbbd5be59f1722b74c38eac9