pep8 cleaning + remove unneeded imports