Fixed rosenbrock.py