Add some code for week2