Changed close button to just close