diff --git a/README.md b/README.md
index 91a53cd..fa2cd58 100644
--- a/README.md
+++ b/README.md
@@ -4,12 +4,7 @@
 
 <br>
 
-[**Installation**](#installation)
-|[**Description**](#description)
-|[**Quick Example**](#quick_example)
-|[**StatefulComputation**](#stateful_computation)
-|[**Benchamrks**](#more)
-|[**Acknowledgements**](#acknowledgements)
+Nested tree tools in python
 
 ![Tests](https://github.com/ASEM000/sepes/actions/workflows/test_default.yml/badge.svg)
 ![Tests](https://github.com/ASEM000/sepes/actions/workflows/test_jax.yml/badge.svg)
diff --git a/docs/_static/logo.svg b/docs/_static/logo.svg
index ea1edbe..801eb22 100644
--- a/docs/_static/logo.svg
+++ b/docs/_static/logo.svg
@@ -1,4 +1,4 @@
-<svg width="829" height="303" viewBox="0 0 829 303" fill="none" xmlns="http://www.w3.org/2000/svg">
-<circle cx="742" cy="181" r="87" fill="#BED9B2"/>
-<path d="M124.764 141.68L95.4675 144.876C94.6389 141.917 93.1889 139.135 91.1174 136.531C89.1051 133.927 86.3825 131.826 82.9498 130.228C79.517 128.63 75.3148 127.831 70.3432 127.831C63.6552 127.831 58.0326 129.281 53.4753 132.181C48.9772 135.081 46.7578 138.839 46.8169 143.456C46.7578 147.421 48.2078 150.647 51.1671 153.133C54.1855 155.619 59.1571 157.661 66.0819 159.259L89.3418 164.23C102.244 167.012 111.832 171.421 118.106 177.458C124.439 183.495 127.635 191.396 127.694 201.162C127.635 209.744 125.119 217.32 120.148 223.889C115.235 230.4 108.4 235.49 99.6401 239.159C90.8806 242.829 80.8191 244.663 69.4554 244.663C52.7651 244.663 39.33 241.171 29.15 234.188C18.9701 227.144 12.9036 217.349 10.9505 204.802L42.2892 201.783C43.7097 207.939 46.7282 212.585 51.3446 215.722C55.9611 218.858 61.9685 220.427 69.3667 220.427C77.0016 220.427 83.1273 218.858 87.7438 215.722C92.4195 212.585 94.7573 208.708 94.7573 204.092C94.7573 200.185 93.248 196.96 90.2296 194.415C87.2703 191.87 82.6538 189.917 76.3801 188.555L53.1202 183.673C40.0402 180.95 30.3633 176.363 24.0897 169.912C17.816 163.402 14.7087 155.175 14.7679 145.232C14.7087 136.827 16.9874 129.547 21.6039 123.392C26.2795 117.178 32.7604 112.384 41.0463 109.01C49.3915 105.577 59.0092 103.861 69.8993 103.861C85.8794 103.861 98.4564 107.264 107.63 114.07C116.863 120.877 122.575 130.08 124.764 141.68ZM195.814 244.663C182.142 244.663 170.334 241.822 160.391 236.141C150.507 230.4 142.902 222.291 137.575 211.815C132.248 201.28 129.585 188.881 129.585 174.617C129.585 160.59 132.248 148.28 137.575 137.685C142.961 127.032 150.478 118.746 160.125 112.827C169.772 106.85 181.106 103.861 194.127 103.861C202.531 103.861 210.462 105.222 217.92 107.945C225.436 110.608 232.065 114.751 237.806 120.374C243.606 125.996 248.163 133.158 251.478 141.858C254.792 150.499 256.449 160.797 256.449 172.753V182.607H144.677V160.945H225.643C225.584 154.79 224.252 149.315 221.648 144.521C219.044 139.668 215.404 135.851 210.729 133.069C206.112 130.287 200.726 128.896 194.571 128.896C188.001 128.896 182.231 130.494 177.259 133.69C172.287 136.827 168.411 140.97 165.629 146.119C162.907 151.209 161.516 156.802 161.456 162.898V181.808C161.456 189.739 162.907 196.545 165.807 202.227C168.707 207.85 172.761 212.17 177.969 215.189C183.178 218.148 189.274 219.628 196.258 219.628C200.933 219.628 205.165 218.977 208.953 217.675C212.741 216.313 216.026 214.331 218.807 211.727C221.589 209.122 223.69 205.897 225.111 202.05L255.118 205.423C253.224 213.354 249.613 220.279 244.287 226.197C239.019 232.057 232.272 236.614 224.045 239.869C215.818 243.065 206.408 244.663 195.814 244.663ZM264.888 293.136V105.636H296.493V128.186H298.357C300.014 124.872 302.352 121.35 305.371 117.621C308.389 113.834 312.473 110.608 317.622 107.945C322.771 105.222 329.341 103.861 337.331 103.861C347.866 103.861 357.365 106.554 365.829 111.94C374.352 117.266 381.099 125.168 386.07 135.643C391.101 146.06 393.616 158.844 393.616 173.996C393.616 188.97 391.16 201.695 386.248 212.17C381.335 222.646 374.647 230.636 366.184 236.141C357.72 241.645 348.132 244.397 337.42 244.397C329.607 244.397 323.126 243.095 317.977 240.491C312.828 237.887 308.685 234.75 305.548 231.08C302.471 227.352 300.074 223.83 298.357 220.516H297.026V293.136H264.888ZM296.404 173.818C296.404 182.637 297.647 190.361 300.133 196.989C302.678 203.618 306.318 208.797 311.053 212.526C315.847 216.195 321.647 218.03 328.453 218.03C335.555 218.03 341.504 216.136 346.298 212.348C351.092 208.501 354.702 203.263 357.129 196.634C359.614 189.946 360.857 182.341 360.857 173.818C360.857 165.355 359.644 157.838 357.217 151.268C354.791 144.699 351.18 139.55 346.386 135.821C341.592 132.092 335.615 130.228 328.453 130.228C321.588 130.228 315.758 132.033 310.964 135.643C306.17 139.254 302.53 144.314 300.044 150.825C297.617 157.335 296.404 165 296.404 173.818ZM462.757 244.663C449.085 244.663 437.278 241.822 427.335 236.141C417.451 230.4 409.845 222.291 404.518 211.815C399.192 201.28 396.528 188.881 396.528 174.617C396.528 160.59 399.192 148.28 404.518 137.685C409.904 127.032 417.421 118.746 427.068 112.827C436.715 106.85 448.049 103.861 461.07 103.861C469.475 103.861 477.406 105.222 484.863 107.945C492.38 110.608 499.008 114.751 504.749 120.374C510.549 125.996 515.107 133.158 518.421 141.858C521.736 150.499 523.393 160.797 523.393 172.753V182.607H411.621V160.945H492.587C492.527 154.79 491.196 149.315 488.592 144.521C485.987 139.668 482.348 135.851 477.672 133.069C473.055 130.287 467.67 128.896 461.514 128.896C454.945 128.896 449.174 130.494 444.202 133.69C439.231 136.827 435.354 140.97 432.572 146.119C429.85 151.209 428.459 156.802 428.4 162.898V181.808C428.4 189.739 429.85 196.545 432.75 202.227C435.65 207.85 439.704 212.17 444.913 215.189C450.121 218.148 456.217 219.628 463.201 219.628C467.877 219.628 472.108 218.977 475.896 217.675C479.684 216.313 482.969 214.331 485.751 211.727C488.532 209.122 490.634 205.897 492.054 202.05L522.061 205.423C520.167 213.354 516.557 220.279 511.23 226.197C505.963 232.057 499.215 236.614 490.989 239.869C482.762 243.065 473.351 244.663 462.757 244.663ZM639.608 141.68L610.311 144.876C609.483 141.917 608.033 139.135 605.961 136.531C603.949 133.927 601.226 131.826 597.794 130.228C594.361 128.63 590.159 127.831 585.187 127.831C578.499 127.831 572.876 129.281 568.319 132.181C563.821 135.081 561.602 138.839 561.661 143.456C561.602 147.421 563.052 150.647 566.011 153.133C569.029 155.619 574.001 157.661 580.926 159.259L604.186 164.23C617.088 167.012 626.676 171.421 632.95 177.458C639.283 183.495 642.479 191.396 642.538 201.162C642.479 209.744 639.963 217.32 634.992 223.889C630.079 230.4 623.243 235.49 614.484 239.159C605.724 242.829 595.663 244.663 584.299 244.663C567.609 244.663 554.174 241.171 543.994 234.188C533.814 227.144 527.747 217.349 525.794 204.802L557.133 201.783C558.553 207.939 561.572 212.585 566.188 215.722C570.805 218.858 576.812 220.427 584.21 220.427C591.845 220.427 597.971 218.858 602.588 215.722C607.263 212.585 609.601 208.708 609.601 204.092C609.601 200.185 608.092 196.96 605.073 194.415C602.114 191.87 597.498 189.917 591.224 188.555L567.964 183.673C554.884 180.95 545.207 176.363 538.933 169.912C532.66 163.402 529.552 155.175 529.612 145.232C529.552 136.827 531.831 129.547 536.448 123.392C541.123 117.178 547.604 112.384 555.89 109.01C564.235 105.577 573.853 103.861 584.743 103.861C600.723 103.861 613.3 107.264 622.474 114.07C631.707 120.877 637.418 130.08 639.608 141.68Z" fill="#AEAEAE"/>
+<svg width="832" height="303" viewBox="0 0 832 303" fill="none" xmlns="http://www.w3.org/2000/svg">
+<circle cx="745" cy="173" r="87" fill="#BED9B2"/>
+<path d="M124.764 141.68L95.4675 144.876C94.6389 141.917 93.1889 139.135 91.1174 136.531C89.1051 133.927 86.3825 131.826 82.9498 130.228C79.517 128.63 75.3148 127.831 70.3432 127.831C63.6552 127.831 58.0326 129.281 53.4753 132.181C48.9772 135.081 46.7578 138.839 46.8169 143.456C46.7578 147.421 48.2078 150.647 51.1671 153.133C54.1855 155.619 59.1571 157.661 66.0819 159.259L89.3418 164.23C102.244 167.012 111.832 171.421 118.106 177.458C124.439 183.495 127.635 191.396 127.694 201.162C127.635 209.744 125.119 217.32 120.148 223.889C115.235 230.4 108.4 235.49 99.6401 239.159C90.8806 242.829 80.8191 244.663 69.4554 244.663C52.7651 244.663 39.33 241.171 29.15 234.188C18.9701 227.144 12.9036 217.349 10.9505 204.802L42.2892 201.783C43.7097 207.939 46.7282 212.585 51.3446 215.722C55.9611 218.858 61.9685 220.427 69.3667 220.427C77.0016 220.427 83.1273 218.858 87.7438 215.722C92.4195 212.585 94.7573 208.708 94.7573 204.092C94.7573 200.185 93.248 196.96 90.2296 194.415C87.2703 191.87 82.6538 189.917 76.3801 188.555L53.1202 183.673C40.0402 180.95 30.3633 176.363 24.0897 169.912C17.816 163.402 14.7087 155.175 14.7679 145.232C14.7087 136.827 16.9874 129.547 21.6039 123.392C26.2795 117.178 32.7604 112.384 41.0463 109.01C49.3915 105.577 59.0092 103.861 69.8993 103.861C85.8794 103.861 98.4564 107.264 107.63 114.07C116.863 120.877 122.575 130.08 124.764 141.68ZM197.064 244.663C183.392 244.663 171.584 241.822 161.641 236.141C151.757 230.4 144.152 222.291 138.825 211.815C133.498 201.28 130.835 188.881 130.835 174.617C130.835 160.59 133.498 148.28 138.825 137.685C144.211 127.032 151.728 118.746 161.375 112.827C171.022 106.85 182.356 103.861 195.377 103.861C203.781 103.861 211.712 105.222 219.17 107.945C226.686 110.608 233.315 114.751 239.056 120.374C244.856 125.996 249.413 133.158 252.728 141.858C256.042 150.499 257.699 160.797 257.699 172.753V182.607H145.927V160.945H226.893C226.834 154.79 225.502 149.315 222.898 144.521C220.294 139.668 216.654 135.851 211.979 133.069C207.362 130.287 201.976 128.896 195.821 128.896C189.251 128.896 183.481 130.494 178.509 133.69C173.537 136.827 169.661 140.97 166.879 146.119C164.157 151.209 162.766 156.802 162.706 162.898V181.808C162.706 189.739 164.157 196.545 167.057 202.227C169.957 207.85 174.011 212.17 179.219 215.189C184.428 218.148 190.524 219.628 197.508 219.628C202.183 219.628 206.415 218.977 210.203 217.675C213.991 216.313 217.276 214.331 220.057 211.727C222.839 209.122 224.94 205.897 226.361 202.05L256.368 205.423C254.474 213.354 250.863 220.279 245.537 226.197C240.269 232.057 233.522 236.614 225.295 239.869C217.068 243.065 207.658 244.663 197.064 244.663ZM267.388 293.136V105.636H298.993V128.186H300.857C302.514 124.872 304.852 121.35 307.871 117.621C310.889 113.834 314.973 110.608 320.122 107.945C325.271 105.222 331.841 103.861 339.831 103.861C350.366 103.861 359.865 106.554 368.329 111.94C376.852 117.266 383.599 125.168 388.57 135.643C393.601 146.06 396.116 158.844 396.116 173.996C396.116 188.97 393.66 201.695 388.748 212.17C383.835 222.646 377.147 230.636 368.684 236.141C360.22 241.645 350.632 244.397 339.92 244.397C332.107 244.397 325.626 243.095 320.477 240.491C315.328 237.887 311.185 234.75 308.048 231.08C304.971 227.352 302.574 223.83 300.857 220.516H299.526V293.136H267.388ZM298.904 173.818C298.904 182.637 300.147 190.361 302.633 196.989C305.178 203.618 308.818 208.797 313.553 212.526C318.347 216.195 324.147 218.03 330.953 218.03C338.055 218.03 344.004 216.136 348.798 212.348C353.592 208.501 357.202 203.263 359.629 196.634C362.114 189.946 363.357 182.341 363.357 173.818C363.357 165.355 362.144 157.838 359.717 151.268C357.291 144.699 353.68 139.55 348.886 135.821C344.092 132.092 338.115 130.228 330.953 130.228C324.088 130.228 318.258 132.033 313.464 135.643C308.67 139.254 305.03 144.314 302.544 150.825C300.117 157.335 298.904 165 298.904 173.818ZM466.507 244.663C452.835 244.663 441.028 241.822 431.085 236.141C421.201 230.4 413.595 222.291 408.268 211.815C402.942 201.28 400.278 188.881 400.278 174.617C400.278 160.59 402.942 148.28 408.268 137.685C413.654 127.032 421.171 118.746 430.818 112.827C440.465 106.85 451.799 103.861 464.82 103.861C473.225 103.861 481.156 105.222 488.613 107.945C496.13 110.608 502.758 114.751 508.499 120.374C514.299 125.996 518.857 133.158 522.171 141.858C525.486 150.499 527.143 160.797 527.143 172.753V182.607H415.371V160.945H496.337C496.277 154.79 494.946 149.315 492.342 144.521C489.737 139.668 486.098 135.851 481.422 133.069C476.805 130.287 471.42 128.896 465.264 128.896C458.695 128.896 452.924 130.494 447.952 133.69C442.981 136.827 439.104 140.97 436.322 146.119C433.6 151.209 432.209 156.802 432.15 162.898V181.808C432.15 189.739 433.6 196.545 436.5 202.227C439.4 207.85 443.454 212.17 448.663 215.189C453.871 218.148 459.967 219.628 466.951 219.628C471.627 219.628 475.858 218.977 479.646 217.675C483.434 216.313 486.719 214.331 489.501 211.727C492.282 209.122 494.384 205.897 495.804 202.05L525.811 205.423C523.917 213.354 520.307 220.279 514.98 226.197C509.713 232.057 502.965 236.614 494.739 239.869C486.512 243.065 477.101 244.663 466.507 244.663ZM644.608 141.68L615.311 144.876C614.483 141.917 613.033 139.135 610.961 136.531C608.949 133.927 606.226 131.826 602.794 130.228C599.361 128.63 595.159 127.831 590.187 127.831C583.499 127.831 577.876 129.281 573.319 132.181C568.821 135.081 566.602 138.839 566.661 143.456C566.602 147.421 568.052 150.647 571.011 153.133C574.029 155.619 579.001 157.661 585.926 159.259L609.186 164.23C622.088 167.012 631.676 171.421 637.95 177.458C644.283 183.495 647.479 191.396 647.538 201.162C647.479 209.744 644.963 217.32 639.992 223.889C635.079 230.4 628.243 235.49 619.484 239.159C610.724 242.829 600.663 244.663 589.299 244.663C572.609 244.663 559.174 241.171 548.994 234.188C538.814 227.144 532.747 217.349 530.794 204.802L562.133 201.783C563.553 207.939 566.572 212.585 571.188 215.722C575.805 218.858 581.812 220.427 589.21 220.427C596.845 220.427 602.971 218.858 607.588 215.722C612.263 212.585 614.601 208.708 614.601 204.092C614.601 200.185 613.092 196.96 610.073 194.415C607.114 191.87 602.498 189.917 596.224 188.555L572.964 183.673C559.884 180.95 550.207 176.363 543.933 169.912C537.66 163.402 534.552 155.175 534.612 145.232C534.552 136.827 536.831 129.547 541.448 123.392C546.123 117.178 552.604 112.384 560.89 109.01C569.235 105.577 578.853 103.861 589.743 103.861C605.723 103.861 618.3 107.264 627.474 114.07C636.707 120.877 642.418 130.08 644.608 141.68Z" fill="#AEAEAE"/>
 </svg>