Grid graphics update

Resolves #3 and fixes #4