success prob_2
refactoring prob_2