STY: PEP8 and docstring fix of grid.py

Pull request is good to merge with these changes.