update code for demo