updated code
modified