#3 Added a restart button