[wip] Add grid for images