Clean up the script