added grid.py and graphs