Update align.rs