add images and clean the code a bit