Adds restart button