clean up based on pylint