Updated wrapping code