Add support to specify to which model the attributes belong