Clean-up the script