Added support for prime numbers