Add dry run option