Updated scripts for figure creation