Add in restore button for diffs