Improve psi tree