Add optional close button