def load_initial_grid(state_slice, dimensions):
    """Make initial grid from the state_slice depending on number of
    dimensions."""
    grid = []
    initial_slice = [
        [column for column in row]
        for row in state_slice
    ]
    if dimensions == 4:
        initial_slice = [initial_slice]
    grid.append(initial_slice)
    return grid