adding missing relationship in models