Converted test program into scalatest test suite