Week 6 Updates

Updated the TODO sections to complete week 6 assignment