Add close modal functionality