Re-implement task reset functionality