Keep code in the "setup" until button press?

Well you can if-else-if-else-if-else till you run out of space,

the idea is that if you have mutually exclusive conditions then why test for them all? If they're not then don't do that.
Sometimes switch-case or another conditional is the way to go... these are things that make arranging logic cleaner, faster executing and less ambiguous.

I wrote code in 78-79 on programmable calculators. One way to know the value of these things is to not have a full set, to learn from what you wish you had and have to make up for. You won't forget soon after that! The TI-56 didn't even have indirect addressing, talk about pain!