added rm_short_traj.py
added figure