add images and clean up UI a little bit