Use return instead of exit.
Add command flags.