Move main CLI execution into a __main__ check