Update week2.hs