added basic data association
added visualizations, fixed weight bug