added enanched visualization function
rewrite trajectories generation functions