Add unit function
Align indentation