Fixing pylint errors