Add a few comments + polishing
Change benchmark to a slightly more complicated one