Visualization tool script modified