PyLint errors fixed