Cleanup

Fix a bunch of pylint warnings.