Add some new code for visualizing histograms