modified code
updated