Modify demo code.