Add collocation sorting