modify wrs_download.py