add belongs_to to draft