Finished belongs_to functiolity