Improve code formatting
Improve code formatting