Removed extra demo code