Updates to figures and references