https://tutorial.eyehunts.com/python/python-choice-input-example-code/