Models setup

relationships are working