Add scripts for figure generation