Update sol2.hs