Add some more gravity tree functions