Rust code optimization