Adding relationship model and tests