Added functionality to restart button