Update to add close button

 Update to add close button