Fixing html output
Done messing around the html, will generate image instead?