How to add elements to a set in Python