Use a set

It's more efficient for checking belonging.
WIP