Added the ability to add comments to the ui