Polished Test.hs