Added restart method