How to create a game over restart button in swift