Got the model associations to work