added scalatest to write some new tests