added more functionality for the output of different class sizes