Adapt script to new Lorentz vector class