Fixed indentation and removed unnecessary import