Add schema to model