Add roles to the data model