Add option to choose belongs_to relationship with roles