Update code example for live demo