Cleans up some pep8 mistakes