Worked more on the visualize file