Small adjustments - as during lection 3