Add demo code for mock