Update code example now that discover() returns a set

Apologies for previous PR, it occurred to me that this is a better way to do things rather than use a list comprehension.