Remove unused imports

Use pylint to find unused imports and remove them.