Scalaで素数判定プログラムを書いてみた

Scalaで素数判定

scalaで入力された数字が素数か判定するプログラムを書いてみました。
素数なら"prime number"が、素数でないなら"not prime number"が出力されます。
Scala勉強中なので改善点があれば教えてください。

PrimeNumber.scala

object PrimeNumber {
  def main(args: Array[String]) = {
    println("Enter number")
    val num: Int = scala.io.StdIn.readInt
    primeNumberChecker(num) match {
      case 0 => println("prime number")
      case 1 => println("not prime number")
      case _ => println("error")
    }
    def primeNumberChecker(x: Int): Int = {
      var check: Int = 0
      if(x > 1) {
        for(i <- 1 to x) {
          if( (i != 1) && (i != x) && (x % i == 0) ) { check = 1 }
        }
      } else { check = 1 }
      check
    }
  }
}