Updated plotting script for my tool