ENH: Make it easier to modify number of classes.