adding work through lecture 20