How To Get Input For Set In Python