minor change on probability and add ques code