Add command-line processing