Make the remove and schedule calls remove true when the task finishes
successfully and false in case of timeout.

Updated docs to reflect the changes.