Add the dry run mode