Move class tree logic switching handler to into Tree controller.

This way we can also easily remember the state of showPrivate
when switching between logics.