Add solution for checking if binary tree is balanced. Can be futher optimized, though.