Remove double import; fix pep8