Minor cleanup of script.