cleaning up code, ran pep8