function note(el)
    el.classes = pandoc.List()
    el.classes:insert("c-note")
    return el
end

function figure(el)
    local images = pandoc.List()

    pandoc.walk_block(el, { Image = function(iel) images:insert(iel) end })

    return pandoc.RawBlock("html",
        "<figure class=\"c-figure\">\
            <a class=\"c-figure__link\" href=\"" .. images[1].src .. "\">\
                <img class=\"c-figure__image\" src=\"" .. images[1].src .. "\" title=\"" .. images[1].title .. "\" />\
            </a>\
        </figure>")
end

function Div(el)
    if el.classes:includes("note") then
        el = note(el)
    elseif el.classes:includes("figure") then
        el = figure(el)
    end

    return el
end