Update getTraj_simple.py