Update main.rs