Simplify the CLI set_logging logic