Update fitness_script.py