Update code to compile and Cargo.toml