Added model class to generation code template