Slight modification of script