Removing demo code