Updating demo code