How can the toolbox be extended to support other types of trajectory prediction datasets, such as those collected from on-board vehicle sensors