Cleaned main.rs