add tests and initial relationship model