Update key2sortedSet.py