Whoops I forgot to also update figure 5.