Added small ui changes. Added extra images for future work.