updated grid.py