Tidied up heatmap drawing

- removed inaccurate comment
- added spaces around arithmetic operators