Add handle_command_args function