Add chapter4 changes