Updates to assignment 3 and lesson 4