Add wrap() method