Improved code styling.