Update main.go

cleanup