Have things working with the labelled data association