Write a Scala program to split a given list into two lists

object Scala_List {    
  def split_list[A](nums: List[A], n: Int): (List[A], List[A]) = {
    (nums.take(n), nums.drop(n))
 }
  
  def main(args: Array[String]): Unit = {
     val nums1 = List(1,2,3,4,5,6)
     println("Original List:")
     println("Split the said list into 2 lists:")
     println(split_list(nums1, 2))
     println(split_list(nums1, 3))
    }
}