Add some images and fix some code