Added the ability to restart game