Update main.py

[theonlykingpin]

Less code using Walrus operator.

[funkyfuture]

there's no use of the symbol choice at all. look out for more issues in this tiny module. they exist.