Fix heat capacity test

Add flag to control histogram printing in PT.
Should fix failling heat capacity test.