updated hw2 script