added rest of my script