Oops. The last line should be

```python
k1 = next(itertools.dropwhile(lambda x: x in col_used, itertools.count()))
```