Fix indent and unefficient code