heatbugs matplotlib plot working