Added key input for restart or quit