Get restart game working