updated assignment1 script