Added click to restart