Add expected attributes to model so that everything works together