Update hw5.hs