Moved where we balance tree. Needed to be BEFORE the while loop.