Selected entity now properly highlights in tree

Fix for: https://github.com/cwrc/CWRC-Writer/issues/174