Converted to use scalatest instead of manual runner
problem 8