Add mode to demo