Update code on demo