Partially Working Code for Week 2