Finished problem 1 script