refactor colocation graph