Modified test.hs