modify test.hs