Set vs dictionary in python