Added ability to restart or quit after winning