Added some features

- Restart game