Jet rapidity calc function updated