Have updated the code

Have updated the code