add some code for demo