Update class-04.hs