Add `output_location` argument