Update README. Also added a visualization image