Currently the implementation of box_wrap is wrong. The box size is b - a and not simply b. As we are always assuming a == 0, the code works fine. Here I am removing the parameter a. If you prefer, I can also simply update the box size in the wrap.