Removed unused import and pep8