Add appropriate divs and classes