Added method for getting capacity.