MOD: final modifications

Removing the need to specify the grid_shape since this can be easily
determined from the input grid_dimensions. I have also removed the
documentation for grid_shape.