Fixed test cases for ass2/prob2