Week2 code added