add missing import
pep8 and pyflakes cleanup