Add unit test via scalatest