Tweak to one of the demo programs