Implemented restart on click