Latest updates to script - commiting code before adding the final changes to make top quality plots
for the paper 01