cpp20 wap to print whether the entered no. is prime or not

#include <iostream>
#include<cmath>

using namespace std;

int main()
{
    int n;
    cin>>n;
    bool flag=0;
    for(int i=2;i<sqrt(n);i++){
        if(n%i==0){
            cout<<"no. is non prime"<<endl;
            flag=1;
            break;
        }
        
        
    }
    if(flag==0){
        cout<<"no. is prime"<<endl;
    }
    

    return 0;
}