Cleaned up the figure captions