Modified the construction of the Co-occurrence graph