Returned to a 2 classes model.