Week 3 Day 1 code