Add functionality for plotting 3D scatter/heat map.