Minor changes to trajectory visualisation