Corrected final_strategy in plot.py