Week5 code added