Update problem1.hs