Switched from lists to sets.

Signed-off-by: Jason Lewis <jason.lewis1991@gmail.com>
Fixing tests after switching to sets.

Signed-off-by: Jason Lewis <jason.lewis1991@gmail.com>