Add align function