modification

modification to my script