updated test.hs