adding work from w1 lectures