Make heat map height configurable