We report progress towards computing the heavy quark momentum diffusion coefficient from the lattice correlator of two chromoelectric fields attached to a Polyakov loop in pure SU(3) gauge theory. Using a multilevel algorithm and tree-level improvement, we study the behavior of the diffusion coefficient as a function of temperature in the wide range $1.1 < T/Tc < 10^4$.