Correction in ParticleTree class