add scripts for plotting SWE