Added the image and fixed a comment in main