added container div and put button inside