Updated restarting game function.