Partial commit - add relationship model