Add classes to model