Added restart and results buttons