Added initial belongsTo functionality