Fix: change WeightedVertex set to set of IDs (because of fucking magic)