Update USMtest.py