Display All elements from array

object hello {
def main(array: Array[String]): Unit = {

val array = Array("Pratik", "shubham", "sujot", "siddesh")

println("Enter a String : ")
val str = scala.io.StdIn.readLine()

val result = array.filter(c => c.contains(str))

if (result.isEmpty) {
  println("No elements found containing the given string.")
} else {
  println("Elements containing the given string : ")
  result.foreach(println)
}

}
}