add classes for model