Add relationships between models
Haven't tested but try something like this