Added Close button

Cleaned up and added a Close button to appear after Start button clicked