Updated to clear the screen upon closing