Add model relations

Just extend the model if needed.