Finished problem, checking if a binary tree is balanced