minor modification to week2.