Improvements to KS Model Plotting script