Multi-class BART Model Assistance

CC @aloctavodia