Fix ugly markup/code