Update README.md
Add images
Remove unecessary code
Minor tweaks and fixes