Update main.go imports for new file structure.