Added tests for tree and heap