Clean up pylint