Update code which generates demo