Define probability mass function (#840)