modify code 2