ran pylint, cleaned up code