tweak forwad log prob printing