Add option for input and output dir