Updated the demo code.

This should've been done a while ago, to keep the two in sync.