Updates with lesson 5 homework