WIP - define belongs_to relation