- cleaned up the script