Double list comprehension is brutal :) . I know what you are doing here, but I'd advise to either do