Updates to figures