Fixing code identation