Add flag for dry running