Remove code from demo file