Remove demo functionality