Draws heat legend.
Cleaned some code (removed save state code).