Fix pylint errors

Shouldn't be controversial...