Week 3 project fix