Improved the code indentation