Update codechef.py