Add missing associations in models