[IMP] Use set, not list generator to generate set