Integrate energy storage