Added close button back in