Add functionailty for user input to exit program