Changed files to work with pylint