For the prime challenge