Fixes tree node highlighting
Merge pull request #32 from chughes87/master

Fixes tree node highlighting