Refactor things to not depend on a tree node structure (only allows chain configurations now but that's ok)