Updated main.go.