Removed unused imports, applied pep8 configuration