Updated code

Updating