add a restart button + better looking on 600x800