Help me in solving LPYAS152 problem

My issue

Wrong Answer: Failed on a hidden test case

My code

import java.util.*;
import java.lang.*;
import java.io.*;

class Codechef
{
	public static void main (String[] args) throws java.lang.Exception
	{
		// your code goes here
		int count=0;
		Scanner sc = new Scanner(System.in);
		int n=sc.nextInt();
	    if((n*n+1)%2==0)
	    {
	        System.out.println("prime");
	    }
	    else
	    {
	        System.out.println("not prime");
	    }

	}
}

Learning course: Roadmap to 3*
Problem Link: https://www.codechef.com/learn/course/klu-roadmap-3star/KLURMP300B/problems/LPYAS152