Added images and changed code for the menu