Add input to demo