lots of changes for heatmap, changes for general plotting for JStokes