simplify main.rs