Added script for plotting