Used autopep8 to clean up code