adding missing models associations