Add classes to model and controller