fixed import and pep8 updates