Fixing code indentation