Added restart game button