Better sizing for battery