Added ability to close modal