Another tweak to one of the demo programs