Add update function to task repo