adding test cases for Prob0, Prob1