Added a plotting script, does not currently work, but is a good starting point