Added helper script for plotting