Code annotation now works on tree nodes