Add more demo code