add labels arg to segment_model