Add week1 code