Added more to relationship model