Finish the visualization part