Adding logic to auto expand prob_num.