Improve visibility of comment tool in tree viewer

Addresses #327, but maybe not enough..?