Update grid.py

swaaaaag