Add documentation for probability