how to write code for check prime number