Update problem02.hs