Made restart button functional