A modified Ehrenfest method that achieves Boltzmann quantum state populations