Change method name and fix the prime method use