Remove demo code