Added function to plot figures needed for paper draft