Update stack_plot.py