Fix imports and reformat code.