FIX: use save_dir for save_files and set_alpha_mask as well