Update q2.hs