add restart game function