Changed the HTML to just have the container div