Implement restart/update functionality