Added draw_trj_evo, other plotting functionality. Plus other minor changes