Add references models to setup associations