drain_done_set.py: Include prefix in filename