Nice, looks like I need to read a bit about itertools!