Improve src/main.rs

[mapx]

Processed with cargo clippy, lines are more succinctly expressed.

[aylei]

Thanks!