Water heater with heat pump