Improved minor readability issue for python script