minor changes to figures