Add ability to declare in which model the attributes belong