Minor code cleanup
Minor code cleanup with pylint