Included more math to solving for the tree level.