Add argument parsing for commands