Rewrite in more efficienct way
Add missing pt and eta checks for probe jet
Fix inner loop