A few changes to improve upon the script: