small update to get_cells()