Addded restart button