Quick solution for wrapping in div

Not really tested (sorry). Some dry coding. But you see what I mean. Hope you find it useful.