fix minor issues with figures