prime/not prime functionality works