added the visualization part of paper