Add option to save histograms to root file