Fix the rooted tree height function

Was returning a sequence of all heights and the heights were all +1.