Updated code
Fixed code formatting