Small changes to figures