Simplify by adding case class Diff
Remove braces