minor changes and figure