Updating demo code.