It's called multinomial logistic regression if there are more than 2 classes. It works exactly how you described.