def addAfter[A](l: List[A], i: A, after: A):List[A] = l match {
  case x :: xs => (if (x == i) { x :: after :: Nil } else { x :: Nil }) ::: addAfter(xs, i, after)
  case Nil => Nil
}