Update get_wrs.py