Add prime number detection tests