Scheduling of energy storage