Updated code formatting

Improved code formatting