Set up associations in the models