Flywheel battery energy storage calculation