Update grid.py