add ability to assign labels (wip)