Adding prime number generator assignment