Add relationships to models