FIX: Clarify stop_probability parameter description