Further editing

captions for fig 3