Updated the usage of prob