Clean up code indentation