Updated align code