Update according to task spec (misunderstood slightly)