Added comment for is_prime function