Update greedy_col_variation.py