Made some small changes to the visualization section.