Multiple tree view with All and Not applicable button

Current design of the tree view looks good but not great with UX. I am trying to re-position the NA button and Selection of ALL.

Any suggestion?