Change return type to Array[Int]