Week 02 code added