Inprove getRoots implementation