Fixed PyLint errors