Update k_merge.py