add pt and pz plots