Add test with case class