pep8 cleanup
made a main