Cleaned up get_hists.py