Update assignment3.hs