Update main.rs

fix some comments