PyLint tidy up