Update tests to use ScalaCheck