Modified the run to include output directory