Added ability for a close button :rocket: