update part of code