Update interview question and add check balance tree function inTree structure