Add command-line flag parsing.