modified plotting for jmst paper