Updated code

Updated code