lots of changes for heatMaps and general plotting - update for JStokes