Creating a CLI in Python