pep8 fixes

proper main() handling