adjust model classes