Final updates to week 3 code