Slight modification to script