Add in code to generate diffs