Update optimalStopping.py