Refactor BEMTREE

// cc @tavriaforever @gela-d @tormozz48