Added an explanatory comment to the script.
Make the code more pythonic.