Fix better supervision tree for brain