Small improvments in figures