Updated to use height2(Node) for testing