Trying to write some tests with ScalaTest