UI improvment, add Dialog restart game