Write a Scala program to create a new string taking the first 3 characters of a given string and return the string with the 3 characters added at both the front and back

object scala_basic {
  def test(str1: String): String =
    {
    if (str1.length < 3) str1 * 3
    else str1.substring(0, 3) + str1 + str1.substring(0, 3)
    }
   def main(args: Array[String]): Unit = {
      println("Result: " + test("Scala"));
      println("Result: " + test("abcd"));
      println("Result: " + test("ab"));
      println("Result: " + test("a"));
    }
}