Make save_dir optional