Corrected errors, and improved the visualization.