Add ability to override the close button