Added functionality to html_writer.py