Week3 code added