Add leafs tests