Energy storage battery capacity selection