added prime_number function