A couple of improvements

Hey guys,

I've a made a couple of changes there:
- reworked UI a bit
- highlight attributes on right-click
- fixed crash with empty tree sreenshoting

Feel free to merge them.

KR,
Aliaksei