Capacity Test script changes