Improve wrap()