Refactor after running pylint.