Fix wk5 final plot