add modifyed grid.py