Updated for week 3 project