Small updates to improve clarity of figures