Add the screen shot, and some demo code change