Update prime.hs