Add particle tagging to AABB tree