finished basic multi-class classification