Added references within the models (not sure if this will work)