more tweaks in figures