cos_calc.py modify