Fixed main.rs