Need to work out restart method