Update cook_graves.py