Fixed minor bug in traj names.
Added better docstring.