Improve the handling of the add button component.

Made the following fixes:
 - made the component a div on both unit and container pages
 - don't show the component for public pages
 - wrote unit tests