cleaned up with pylint