Added a second function to test.hs