README edit
added images and refactored code