- Changed the comparison figures to make legend larger
- Added legend to traj-comp figure