Add option to specify output directory