added initial version of visualize script