Updated Visualization Script