Restart button is now functional and so is test for it.