Clean up main.rs a bit.

For some nebulous definition of "clean up."