Since you moved the class capacity part into a new for loop, should this go at the end?