Fix some pylint warnings

Just cleaning up a little