added case class What to make code more readable.