fix the cell classing a little