changed imports to meet pep8