Updated figures and included plotting code