Add reset game code