Minor improvements to probability section