Add capacity constraint function. Need to add to Scheduler obj