Week 3 Updated Code