Update problem3.hs