Minor fix in week 4