Clean up comments in is_prime