clean up wham_potential a bit more