Small improvement in longitudinal_tracker_mpi.py