fixing some bugs with saving nn hists