Visualization code has been added.