Should make things simpler. Also we could use [rich-click](https://github.com/ewels/rich-click/) if we want to be fancy.