Added a return html feature to the class