added test for prob to sum to 1