Update greedy.py