Added close button stuff