Used pylint to clean code