Update doc block too