Made plotting script much easier to use