Add changes for last lesson