How to do if/elif/else for 4 choices?