Update rosenbrock.py