Debugged further problems with set_prob()