Improved commenting in main.rs