Attempt at fixin pylint errors