Add support for specifying the output folder