def menu$\_$choice$()$:

print$("1.$ Sweater"$)$

print$("2.$ Suit"$)$

print$("3.$Exit"$)$

choice $=$ input$("$Enter your choice: $")$

return choice

def main$()$:

while True:

choice $=$ menu$\_$choice$()$

if choice $=="1"$:

print$("$Sweater$")$

elif choice $=="2"$:

print$("$Suit$")$

elif choice $=="3"$:

break

else:

print$("$Invalid choice. Please try again."$)$

main$()$

Can you use one more collections to improve the efficiency of this code?