Added code to capture diff image for demo