Add dependency to scalatest.
Remove irrelevant code.