Write a Scala program to check whether a string 'yt' appears at index 1 in a given string. If it appears return a string without 'yt' otherwise return the original string

object scala_basic {
  def test(str: String): String = { 
      if (str.drop(1).startsWith("yt")) str.replaceFirst("yt", "") 
      else str
    }
     
   def main(args: Array[String]): Unit = {
      println("Result: " + test("Scala"));
      println("Result: " + test("yytade"));
      println("Result: " + test("ytsues"));      
    }
  }