add trajectory saving