Add PlayState restart method, so we can save the level data