Improve energy storage