Update main.go

Modify import paths.