Some enhancements to tools/probs_print.py.