Adding prime test function