Add optional color argument for drawing a grid. Also, fix enpoint artifact.