Updating code module slightly