array of strings and read a string from user

Q4 Create array of strings and read a string from user. Display all the elements of array containing given string. in scala program
object demo
{
def main(args:Array[String])
{
var arrayData = Array("Hello","World","Scala","Programming")
println("Enter a String to search: ")
var searchString =scala.io.StdIn.readLine()
for(data <- arrayData)
{
if(data.contains(searchString))
println(data)
}
}
}