Update heatmap to allow for non-RA types

[maxkw]

Update the heatmap matrix code which is a key input into all evolution functions to accept non-RA types.