add comment to prob2.hs