used pylint to clean up code