Visualization function added