Updated align.go