add test to problem2 and prime