Makes minor improvements to class capacities script.