updated lesson based on feedback from JB