add prime number test code
small fix