Adds a fully working tree_gravity_test