Week6 code added