used autopep8 to clean up the src