Code modification - step 1