Test cases working, need to catch TreeException