Program java : menentukan bilangan prima

Assalamualaikum,

udah pusing mikir kode program java.. sama deh kalau githu.. :D
ane cuma mau bagi program cara menentukan bilangan prima.

neh programnya

import javax.swing.JOptionPane;

public class BilanganPrima {
 public static void main(String[] args) {
  String numStr = JOptionPane.showInputDialog("Masukkan bilangan integer");
  int numInt = Integer.parseInt(numStr);
  boolean flag = true;
  
  if(numInt < 2) {
   flag = false;
  } else if(numInt > 2) {
   for(int i = 2; i <= numInt / 2; i++) {
    if(numInt % i == 0) {
     flag = false;
     break;
    }
   }
  }
  
  String ket = "";
  
  if(flag) {
   ket = " adalah bilangan Prima";
  } else {
   ket = " bukan bilangan Prima";
  }  
  
  JOptionPane.showMessageDialog(null, numInt + ket);
 }
} 

Hasilnya :
setelah menginputkan angka