Adding Visualization API code