Fix indentation, missing import