I've used list comprehensions where they make sense. But it's a non-trivial (although a simple) program, so there are for loops too. So I'm using a mix of chai

I've used list comprehensions where they make sense. But it's a non-trivial (although a simple) program, so there are for loops too. So I'm using a mix of chain and permutations from itertools, list comprehensions and for loops.