Improved code indentation