Update grid.py

I didn't know how to put it in a loop to print out the grid but just extended the gird more.