Adding return to reset function