Week 4 code added