restart game function almost works