added restart button but not totally working