Added more elegant way to restart games