Working on restart button.