Add all figures with labels and captions
Add future work from Chaitu