Add power off charging support