add demo code to the repo