Add tree size and height functios