update code by SW

updated code by SW