minor updates to plotting for case study 2.