DOC: A few adjustments in image_grid.py.