Add auto-labeling support