Updated plotting code to work with output changes from SIMTOI.