Update simulated_annealing.py