minor changes in figure