Add todo for lin-reg