ran autopep8 and cleaned up the code