Small change to heat map test + ToDo list