Updated Assignment lesson 6