Update ch2.hs