Fix indentation and import