Updated code
Fixed code