edited heat map to use mwas functions