Added support for restarting game