Updated code
modified code