Can you test the case where stop_probabilty is equal to 1?